from data_util import load_ORLdataset
from pca import pca_face_recognitioin
from lbp import lbp_face_recognition
from cnn import cnn_face_recognition


if __name__ == '__main__':
    orl_path = "/home/aaron/Datasets/att_faces/"
    # orl_path = "/Users/apple/Desktop/FYP/Datasets/att_faces/"
    grayImg = True
    train_X, train_Y, test_X, test_Y = load_ORLdataset(orl_path, grayImg)

    print('Train data shape: ', train_X.shape)
    print('Train labels shape: ', train_Y.shape)
    print('Test data shape: ', test_X.shape)
    print('Test labels shape: ', test_Y.shape)

    num_components = 50
    pca_face_recognitioin(train_X, train_Y, test_X, test_Y, num_components)

    lbp_face_recognition(train_X, train_Y, test_X, test_Y)

    cnn_face_recognition(train_X, train_Y, test_X, test_Y)














